package com.huangguolin.cloud.spring.util;

import com.huangguolin.cloud.mybatis.ztree.ChildTree;
import com.huangguolin.cloud.mybatis.ztree.ZtreeEntity;

import java.util.ArrayList;
import java.util.List;

public class CloudUtil {
    public static List<ChildTree> changeChildTree(List<ZtreeEntity> list) {
        List<ChildTree> childTrees= new ArrayList<>();
        for(ZtreeEntity ztreeEntity : list){
            childTrees.add(new ChildTree(ztreeEntity));
        }
        //输出的结果ztree列表
        List<ChildTree> rstchildTrees= new ArrayList<>();
        Integer index = -1;
        for(ChildTree childTree :childTrees) {
            index=-1;
            for (ChildTree parentTree : childTrees) {
                if (childTree.getpId().equals(parentTree.getId())) {
                    parentTree.addChildren(childTree);
                    index=0;
                }
            }
            if(index ==-1){
                rstchildTrees.add(childTree);
            }
        }
        return rstchildTrees;
    }
}